numpy matrix functions in Python